package com.example.avmanagerweb.Utils;

import com.example.avmanagerweb.result.exceptionhandler.JWException;
import org.bytedeco.javacv.FFmpegFrameGrabber;
import org.bytedeco.javacv.Frame;
import org.bytedeco.javacv.FrameGrabber;
import org.bytedeco.javacv.Java2DFrameConverter;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.awt.image.RenderedImage;
import java.io.File;
import java.io.IOException;

/**
 * @Author Zhengzx
 * @Date 2022/4/21 16:25:58
 * @Title 截取视频
 * @Version 1.0
 */
public class VideoProcessing {


    /**
     * 获取视频第十二帧
     *
     * @param videoPath 视频地址
     * @param imgPath   生成图片的名字（包含全路径）
     * @param frameNum  截取第几帧
     * @author zhengzx
     * @date 2022/04/21 16:42:14
     */
    public static void getVideoPicture(String videoPath, String imgPath, int frameNum) throws FrameGrabber.Exception {
        File imgFile = new File(imgPath);
        //判断保存的文件的文件夹是否存在，不存在创建。
        if (!imgFile.getParentFile().exists()) {
            System.out.println("保存文件的文件夹不存在，创建。");
            imgFile.getParentFile().mkdirs();
        }
        File videoFile = new File(videoPath);
        if (videoFile.exists()) {
            System.out.println("视频存在：{}" + videoFile);
            //实例化“截取视频”对象
            FFmpegFrameGrabber ff = new FFmpegFrameGrabber(videoFile);
            ff.start();
            int ftp = ff.getLengthInFrames();
            int flag = 0;
            Frame frame = null;
            while (flag <= ftp) {
                //获取帧
                frame = ff.grabImage();
                //过滤前frameNum帧，避免出现全黑图片
                if ((flag > frameNum) && (frame != null)) {
                    break;
                }
                flag++;
            }
            try {
                ImageIO.write(frameToBufferedImage(frame), "jpg", imgFile);
            } catch (IOException e) {
                throw new JWException(400, "存储异常");
            }
            ff.close();
            ff.stop();
        } else {
            throw new JWException(400, "视频不存在：{}" + videoFile);
        }
    }

    /**
     * 帧转为流
     */
    private static RenderedImage frameToBufferedImage(Frame frame) {
        //创建BufferedImage对象
        Java2DFrameConverter converter = new Java2DFrameConverter();
        BufferedImage bufferedImage = converter.getBufferedImage(frame);
        return bufferedImage;
    }
}

